from skimage.metrics import peak_signal_noise_ratio as psnr
from PIL import Image
import numpy as np
import cv2

hr = np.array(Image.open(r'D:'))
lr = np.array(Image.open(r'D:'))
hr_3 = cv2.merge((hr ,hr ,hr ))
lr_3 = cv2.merge((lr ,lr ,lr ))

output_ours = np.array(Image.open(r'D:'))
output_BN = np.array(Image.open(r'D:'))
output_unet = np.array(Image.open(r'D:'))
output_vdsr = np.array(Image.open(r'D:'))

if __name__ == "__main__":
    print("PSNR hr:lr", psnr(hr_3, lr_3), "dB")
    print("")
    print("PSNR hr:output_BN", psnr(hr_3, output_BN), "dB")
    print("")
    print("PSNR hr:ours", psnr(hr_3, output_ours), "dB")
    print("")
    print("PSNR hr:unet", psnr(hr_3, output_unet), "dB")
    print("")
    print("PSNR hr:vdsr", psnr(hr_3, output_vdsr), "dB")